from volcengine.viking_knowledgebase import VikingKnowledgeBaseService

# pip install volcengine aiohttp  deprecated
viking_knowledgebase_service = VikingKnowledgeBaseService(
    host="api-knowledgebase.mlp.cn-beijing.volces.com",
    scheme="https",
    connection_timeout=30,
    socket_timeout=30,
)
viking_knowledgebase_service.set_ak("AKLTYjE2MmRiMmUzODk5NDBkM2E5ZTc5OTQ2MTc0YTgyNDY")
viking_knowledgebase_service.set_sk(
    "T0dWa1pqQTJZVE5rTldKaE5EYzVNV0k0TURFM00yVXhaVEk0TjJVeU5XWQ=="
)
datas = [
    {
        "query": "退改",
        "content": "如果您需要人工服务，可以拨打人工客服电话：4006660921",
        "title": "无",
    },
    {
        "query": "退改",
        "content": "1、1日票 1.5日票 2日票的退款政策： -到访日前2天的00:00前，免费退款 - 到访日前2天的00:00至到访日前夜23:59期间,退款需扣除服务费（人民币80元） - 到访日当天（00:00 之后），不可退款 2、半日票的退款政策： - 未使用的门票可在所选入...",
        "title": "门票退改政策｜北京环球影城的门票退改政策",
    },
    {
        "query": "退改",
        "content": "如果您需要人工服务，可以拨打人工客服电话：4006660921",
    },
]

res = viking_knowledgebase_service.rerank(datas=datas, rerank_model="")
print(res)
